
import logging
import os
from tqdm import tqdm
from omegaconf import OmegaConf
config = OmegaConf.load("configs/train.yaml")

def _55_to_01(img):
    """tensor"""
    return (img / 2 + 0.5).clamp(0, 1)


def create_logger(name=__name__,log_dir='./logs', level=20):
    # 日志
    logger = logging.getLogger(name) # 1 日志器
    logger.setLevel(level=level)

    os.makedirs(log_dir,exist_ok=True)

    fh = logging.FileHandler(f'{log_dir}/{config.log_name}',mode='a') # 2文件处理器
    #sh = logging.StreamHandler() # 2标准流处理器

    formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s',datefmt='%Y-%m-%d %H:%M:%S') # 3 格式化器
    fh.setFormatter(formatter) # 4 格式化器加入处理器
    #sh.setFormatter(formatter) 
    logger.addHandler(fh) # 5 处理器加入日志器
    #logger.addHandler(sh) 
    return logger